Data Analytics with Python assignment answer